package tree.easy;

import tree.TreeNode;

/**
 * 543. 二叉树的直径
 */
public class diameterOfBinaryTree {
//    public int diameterOfBinaryTree(TreeNode root) {
//        if (root == null) {
//            return 0;
//        }
//        int leftHeight = height(root.left);
//        int rightHeight = height(root.right);
//        int r = leftHeight + rightHeight;
//        int childR = Math.max(diameterOfBinaryTree(root.left), diameterOfBinaryTree(root.right));
//        return Math.max(r, childR);
//    }
//
//    public int height(TreeNode root) {
//        if (root == null) {
//            return 0;
//        }
//        return Math.max(height(root.left), height(root.right)) + 1;
//    }

    int max = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        dfs(root);
        return max;
    }

    public int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int left = dfs(root.left);
        int right = dfs(root.right);
        max = Math.max(max, left + right);
        return Math.max(left, right) + 1;
    }

}
